﻿using NMock2;
using Pagination;
using Pagination.Storage;

namespace Tests.Expectations.Pagination
{
    public static class StorageExpectations
    {
        public static void ExpectAdd(this IStorage mock,params object [] elements)
        {
            foreach (var element in elements)
                Expect.Once.On(mock).Method("Add").With((int)element).Will();
         }

        public static void ExpectRange(this IStorage mock, int from, int to)
        {
            Expect.Once.On(mock).Method("Range").With(from, to).Will();
        }
    }
}
